"""
pip config set global.index-url https://pypi.doubanio.com/simple
pip install pymysql

向关系型数据库的二维表中插入数据
"""
import pymysql

no = int(input('部门编号: '))
name = input('部门名称: ')
location = input('部门所在地: ')

# 第一步：建立连接
# host - 主机 - 确定要连接哪一台服务器（IP地址或主机名）
# port - 端口 - 确定连接服务器上的哪个服务（端口用来区分不同的服务）
# user和password - 用户名和口令（提示：不会用超级管理员账号）
# database和charset - 连接的数据库和使用的字符编码方式
conn = pymysql.connect(host='10.7.174.103', port=3306,
                       user='guest', password='Guest.618',
                       database='hrs', charset='utf8mb4',
                       autocommit=True)
try:
    # 第二步：获得游标对象
    with conn.cursor() as cursor:
        # 第三步：通过游标向数据库服务器发出SQL语句，获取执行结果
        affected_rows = cursor.execute(
            'insert into tb_dept (dno, dname, dloc) values (%s, %s, %s)',
            (no, name, location)
        )
        if affected_rows == 1:
            print('添加部门成功！')
    # 第四步：提交（操作成功）
    conn.commit()
except pymysql.MySQLError as err:
    print(err)
    # 第四步：回滚（操作失败）
    conn.rollback()
finally:
    # 第五步：关闭连接（释放资源）
    conn.close()
